# ws/ws.py
from flask_socketio import SocketIO

# 初始化 SocketIO（使用 threading 模式，避免与 ROS2 线程冲突）
socketio = SocketIO(cors_allowed_origins="*")

def init_app(app):
    """将 SocketIO 与 Flask 应用关联。"""
    socketio.init_app(app, async_mode='threading', cors_allowed_origins="*")
    return socketio

@socketio.on('connect')
def handle_connect():
    print("WebSocket 客户端已连接。")

@socketio.on('disconnect')
def handle_disconnect():
    print("WebSocket 客户端已断开连接。")
